package com.lihepeng.leecode.array;

import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/**
 * 今天，书店老板有一家店打算试营业 customers.length 分钟。每分钟都有一些顾客（customers[i]）会进入书店，所有这些顾客都会在那一分钟结束后离开。
 *
 * 在某些时候，书店老板会生气。 如果书店老板在第 i 分钟生气，那么 grumpy[i] = 1，否则 grumpy[i] = 0。 当书店老板生气时，那一分钟的顾客就会不满意，不生气则他们是满意的。
 *
 * 书店老板知道一个秘密技巧，能抑制自己的情绪，可以让自己连续 X 分钟不生气，但却只能使用一次。
 *
 * 请你返回这一天营业下来，最多有多少客户能够感到满意的数量。
 *  
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/grumpy-bookstore-owner
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Solution1052 {
    // 使用滑动窗口
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        List<Integer> unGrumpyAround = new ArrayList<>();
        int res = 0;
        for (int i=0;i<grumpy.length;i++){
            if (grumpy[i] == 0){
                // 不生气
                res+=customers[i];
            }else {
                // 生气
                unGrumpyAround.add(customers[i]);
            }
        }
        // 剩下通过滑动窗口方法窗口长度固定为X
        int max =0;
        for (int i=0;i<unGrumpyAround.size();i++){
            max+=unGrumpyAround.get(i);
        }
        for (int i=X;i<unGrumpyAround.size()-X+1;i++){
            int currentMax = max+unGrumpyAround.get(i)-unGrumpyAround.get(i-unGrumpyAround.size()-X+1);
            max = Math.max(max,currentMax);
        }
        return res+max;
    }

    /**
     * 官方滑动窗口方法
     * @param customers
     * @param grumpy
     * @param X
     * @return
     */
    public int maxSatisfied01(int[] customers, int[] grumpy, int X) {

        return 0;
    }

    @Test
    public void runTest(){
        int []customers = {1,0,1,2,1,1,7,5};
        int []grumpy={0,1,0,1,0,1,0,1};
        int X = 3;
        int i = this.maxSatisfied(customers, grumpy, X);
        System.out.println(i);
    }
}
